import time
import redis
from rediscluster import ClusterBlockingConnectionPool,RedisCluster,ClusterConnectionPool


class RedisQueueCluster(object):
    def __init__(self, redis_conf, redis_key):
        for i in range(5):
            try:
                startup_nodes = [{'host': redis_conf['host'], 'port': redis_conf['port']}]
                pool =  ClusterConnectionPool(startup_nodes=startup_nodes,password="",decode_responses=True)
                self.__db = RedisCluster(connection_pool=pool,decode_responses=True)
                # self.key = '%s:%s' % (namespace, name)
                self.key = redis_key

                self.__db.ping()
                break
            except Exception as e:

                time.sleep(1)
                continue
    def get_all(self):
        item = self.__db.lrange(self.key, 0, -1)
        # self.__db.ltrim(self.key, 0, -1)
        return item
    def qsize(self):
        return self.__db.llen(self.key)  # 返回队列里面list内元素的数量

    def rpush(self, item):
        self.__db.rpush(self.key, item)  # 添加新元素到队列最右方

    def lpush(self, item):
        self.__db.lpush(self.key, item)  # 添加新元素到队列最左方

    def get_wait(self, timeout=None):
        # 返回队列第一个元素，如果为空则等待至有元素被加入队列（超时时间阈值为timeout，如果为None则一直等待）
        item = self.__db.blpop(self.key, timeout=timeout)
        # if item:
        #     item = item[1]  # 返回值为一个tuple
        return item

    def get_nowait(self):
        # 直接返回队列第一个元素，如果队列为空返回的是None
        item = self.__db.lpop(self.key)
        return item

    def sis_member(self, value):
        """
        判断当前 key 对应的 Redis Set 中是否包含某个值。

        :param value: 要查找的值（字符串）
        :return: bool，存在返回 True，否则 False
        """
        return self.__db.sismember(self.key, value)
    def get_cookie(self):
        res = self.__db.get(self.key)
        return res

    def sadd(self, value):
        """
        将一个元素添加到当前 key 对应的 Redis Set 中。

        :param value: 要添加的值（字符串）
        :return: int，返回 1 如果元素被新增，0 表示已存在
        """
        return self.__db.sadd(self.key, value)
    def set_cookie(self, item):
        self.__db.set(self.key, item)

    def del_key(self):
        self.__db.delete(self.key)
    # def get(self):
    #
    #     return self.__db.hkeys(1)
